#ifndef __IMAGE_PROCESS_H__
#define __IMAGE_PROCESS_H__

#include <opencv2/opencv.hpp>
#include "zutils.h"
#include <thread>
#include "bsnn_user_api.h"



#define  IMG_WIDTH          640
#define  IMG_HEIGHT         640
#define  REG_PITCH          256

#define  OUT_CHANNELS       255
#define  PER_ANCHORS_CHN    85
#define  NUM_CLASSES        80

#define CONF_THRESHOLD      0.5f
#define NMS_THRESHOLD       0.5f


void preprocess(const cv::Mat& src, cv::Mat& dst);

void process_output(bsnn_user_output_t *bsnn_output, std::vector<ObjInfo> &result);

void draw_bboxes(cv::Mat& img, const std::vector<ObjInfo>& result);



#endif